package 回溯算法;

import java.util.ArrayList;
import java.util.List;

public class restoreIpAddresses87 {

    public List<String> restoreIpAddresses(String s) {
        int count = 4;
        int[] segments = new int[4];
        List<String> ans = new ArrayList<String>();// 返回的结果
        int segid = 0;
        int index = 0;
        dfs(s, segid, index, segments, ans, count);
        return ans;
    }

    public void dfs(String s, int segid, int index, int[] segments, List<String> ans, int count) {
        // 如果找到了 4 段 IP 地址并且遍历完了字符串，那么就是一种答案
        if (segid == count) {
            if (index == s.length()) {
                StringBuffer ipAddr = new StringBuffer();
                for (int i = 0; i < count; ++i) {
                    ipAddr.append(segments[i]);
                    if (i != count - 1) {
                        ipAddr.append('.');
                    }
                }
                ans.add(ipAddr.toString());
            }
            return;
        }
        // 如果还没有找到 4 段 IP 地址就已经遍历完了字符串，那么提前回溯
        if (index == s.length()) {
            return;
        }
        // 由于不能有前导零，如果当前数字为 0，那么这一段 IP 地址只能为 0
        if (s.charAt(index) == '0') {
            segments[segid] = 0;
            dfs(s, segid + 1, index + 1, segments, ans, count);
        }
        // 一般情况，枚举每一种可能性并递归
        int addr = 0;
        for (int i = index; i < s.length(); ++i) {
            addr = addr * 10 + (s.charAt(i) - '0');
            if (addr > 0 && addr <= 0xFF) {
                segments[segid] = addr;
                dfs(s, segid + 1, i + 1, segments, ans, count);
            } else {
                break;
            }
        }
    }
}
